python while loop example code